package it.polimi.traveldream.dto;

import java.util.HashMap;
import java.util.Map;

public class DetailedDefaultPackageDTO extends DefaultPackageDTO {

	private AccomodationDTO accomodation;

	private TransportDTO going;

	private TransportDTO back;

	private Map<Integer, ScheduleDTO> schedule;

	public DetailedDefaultPackageDTO() {
		super();
		this.accomodation = new AccomodationDTO();
		this.going = new TransportDTO();
		this.back = new TransportDTO();
		this.schedule = new HashMap<>();
	}

	public AccomodationDTO getAccomodation() {
		return this.accomodation;
	}

	public void setAccomodation(AccomodationDTO accomodation) {
		this.accomodation = accomodation;
	}

	public TransportDTO getGoing() {
		return this.going;
	}

	public void setGoing(TransportDTO going) {
		this.going = going;
	}

	public TransportDTO getBack() {
		return this.back;
	}

	public void setBack(TransportDTO back) {
		this.back = back;
	}

	public Map<Integer, ScheduleDTO> getSchedule() {
		return this.schedule;
	}

	public void setSchedule(Map<Integer, ScheduleDTO> schedule) {
		this.schedule = schedule;
	}

	public void addSchedule(ScheduleDTO scheduleDTO) {
		this.getSchedule().put(scheduleDTO.getDay(), scheduleDTO);
	}
}
